#!/usr/env/bin python
# -*- coding: utf-8 -*-

# @Time    : 2020-12-14 6:00 下午
# @Author  : yangdy
# @File    : [中等]1641.统计字典序元音字符串的数目
# @Software: PyCharmCE
from tools.my_methods import *


class Solution:
    def countVowelStrings(self, n: int) -> int:
        dp = [[1 if j == 0 else 0 for i in range(5)]for j in range(n)]
        print(dp)
        for i in range(1, n):
            for j in range(5):
                for k in range(j+1):
                    dp[i][j] += dp[i-1][k]
        return sum(dp[n-1])


questions = [
    (1,),
    (2,),
    (33,),
]

work(questions, lambda q: Solution().countVowelStrings(q[0]))
